const fs = require('fs');
const mime = require('mime-types');
const path = require('path');

exports.fileInit = (req, res, next) => {
	req.mkdir = function (filePath) {
		if (!fs.existsSync(filePath)) {
			fs.mkdirSync(filePath, {
				recursive: true // 是否递归创建目录
			});
		}
	};

	req.getImageFormat = async (filePath) => {
		const ext = path.extname(filePath).slice(1); // 获取文件扩展名（去掉点）
		const mimeType = mime.lookup(ext); // 根据扩展名获取 MIME 类型
		if (mimeType && mimeType.startsWith('image/')) {
			return ext.toUpperCase(); // 返回图片格式（如 JPEG、PNG 等）
		}
	};
	next();
};
